An autoregressive integrated moving average and long short-term memory (ARIM-LSTM) hybrid model for multi-source epidemic data prediction

The COVID-19 pandemic has far-reaching impacts on the global economy and public health. To prevent the recurrence of pandemic outbreaks, the development of short-term prediction models is of paramount importance. We propose an ARIMA-LSTM (autoregressive integrated moving average and long short-term memory) model for predicting future cases and utilize multi-source data to enhance prediction performance. Firstly, we employ the ARIMA-LSTM model to forecast the developmental trends of multi-source data separately. Subsequently, we introduce a Bayes-Attention mechanism to integrate the prediction outcomes from auxiliary data sources into the case data. Finally, experiments are conducted based on real datasets. The results demonstrate a close correlation between predicted and actual case numbers, with superior prediction performance of this model compared to baseline and other state-of-the-art methods.


INTRODUCTION
On July 9, 2020, the World Health Organization (WHO) proclaimed the COVID-19 outbreak a pandemic (Shahid, Zameer & Muneeb, 2020).According to World Health Organization (WHO) (2023) on May 4, 2023, COVID-19 no longer constituted an international public health emergency.Nowadays, COVID-19 has been classified as a Class B infectious disease, and management methods akin to Class A infectious diseases are being implemented (Li et al., 2020).The three-year pandemic caused severe damage to the global economy and people's lives and properties.These devastations underscored that society would do well to invest now in creating prevention models for the next pandemic (Caulkins et al., 2023).
The epidemic development trend prediction has become a central focus for global scholars, involving traditional mathematical and deep learning models.Benvenuto et al. (2020) employed an autoregressive integrated moving average (ARIMA) model to forecast the epidemiological trends.Meanwhile, Iqbal et al. (2021) employed a long short-term memory (LSTM) model to predict the percentage of positive patients in Pakistan for June 2020.Kong (2021) utilized the ARIMA model to interpolate the incidence data of dengue fever in Zhejiang Province over the past 15 years and established the relationship between dengue fever incidence and meteorological factors using the LSTM model.Li, Chen & Yang (2022) proposed a novel hybrid forecasting model named GVMD-ELM-ARIMA, based on Gradient Variational Mode Decomposition (GVMD), extreme learning machine (ELM), and autoregressive integrated moving average (ARIMA), to enhance the prediction accuracy of cumulative COVID-19 confirmed cases.These works demonstrate that hybrid models exhibit superior performance compared to singular traditional mathematical or deep learning prediction models.Traditional mathematical models excel in handling linear prediction tasks with long-term dependencies, such as seasonality.In contrast, deep learning models exhibit significant advantages in capturing more complex nonlinear patterns, particularly in tasks involving short-term fluctuations.The transmission process of pandemics presents characteristics of periodic dependencies and dynamically complex variations, making it challenging for a single prediction model to forecast pandemic trends accurately.Thus, integrating traditional mathematical models with deep learning holds paramount research significance for pandemic trend prediction tasks.
The data sources for epidemic forecasting tasks are diverse, encompassing case data, sentiment text data, and travel mobility data.Tiwari, Kumar & Guleria (2020) utilized confirmed case data in India to predict the number of confirmed cases, cured cases, and deaths.They predicted when the epidemic would peak and made prevention recommendations.Chang & Nguyen (2022) employed flight review data from TripAdvisor to forecast the impact of COVID-19 on the aviation industry.Schlosser et al. (2020) used mobile phone mobility data to predict changes in mobile network structure resulting from lockdowns.The centralization of mobile networks caused the peak of the epidemic outbreak in advance, but it helped reduce the epidemic peak.The privacy concerns surrounding epidemic case data and its scarcity challenge prediction task accuracy.While auxiliary data sources such as sentiment text data and mobility data contain vast amounts of information, they may not directly reflect the dynamics of epidemic spread.Thus, leveraging auxiliary data sources to support case data and undertake prediction tasks on epidemic trends, enhancing prediction accuracy, has emerged as a new challenge.
Therefore, we aim to predict future epidemic trends in a multi-source data scenario.To address this issue, we propose a multi-source data fusion-based ARIMA-LSTM model.First, we forecast future trends based on case data and population mobility data using the ARIMA-LSTM model.Subsequently, we integrate the prediction results of auxiliary data sources into the case data through a fully connected layer based on the Bayes-Attention mechanism.Finally, we experiment with real datasets to validate the model's effectiveness.Our main contributions are threefold: • We propose a pandemic future case trend prediction model based on ARIMA and LSTM, which combines the advantages of both linear and nonlinear models to enhance prediction performance.• We design a fully connected layer based on a Bayes-Attention mechanism to fusion multi-source data, thereby enhancing the accuracy of predictions.

RELATED WORK
Epidemics are one of the most serious issues in public health, characterized by their suddenness and seasonality in transmission (Merow & Urban, 2020).The suddenness leads to nonlinear growth in infection cases.The seasonality causes linear periodic changes in transmission peaks.Hybrid models combining ARIMA and LSTM can handle such linear and nonlinear tasks.Furthermore, the pandemic has prompted governments to implement lockdown measures, resulting in changes in population mobility.Simultaneously, intense population movements due to factors such as holidays have stimulated peaks in the pandemic.Diverse datasets exhibit a notable correlation, as shown in Fig. 1.Multisource data sets can complement each other through such correlations.Therefore, we will introduce related work from hybrid models and multi-source data.

Hybrid models
Researchers have applied various techniques to forecast the epidemic pattern, involving time series models and deep learning models.Time series models such as ARIMA excel in handling linear and periodic tasks.Mahalle et al. (2020) presented medical insights into COVID-19 and used the Prophet model for forecasting.Due to its open-source algorithms, accuracy, and faster data fitting, they recommend Prophet for forecasting.Subsequently, Kumar & Susan (2020) forecasted epidemics in ten countries using ARIMA and Prophet time series prediction models.They demonstrated that the ARIMA model in predicting morbidity rates is more effective.Kumar & Viral (2021) discussed the impact of COVID-19 on the Indian economy, society, and financial sectors using the ARMA model.They illustrated the future scenarios of India under the influence of epidemic in the coming months.Alabdulrazzaq et al. (2021) forecasted the confirmed and recovered cases in various stages of Kuwait's gradual prevention plan using the ARIMA model.Such works all have predicted the future trends of the epidemic.However, the accuracy of the epidemic forecasts is not high.It is unsuitable in complex and dynamic environments.
Deep learning models, such as LSTM, excel at pattern recognition and prediction.Oliveira, Gruetzmacher & Teixeira (2021) utilized artificial neural networks (ANN) to forecast the confirmed cases and death counts for the next 7 days in Brazil, Portugal, and the United States time series.Chandra, Jain & Singh Chauhan (2022) employed recurrent neural networks (RNN) for multi-step short-term infection forecasting.Chimmula & Zhang (2020) used the LSTM model to predict the trends of the epidemic and the likely cessation time.They suggested COVID-19 might cease around June 2020.Kong et al. (2024) proposed a multi-modal traffic forecasting framework, MPGNNFormer, to predict human mobility patterns, addressing issues in both long-term and short-term scenarios.Such works achieved predictions for specific case numbers of the epidemic.However, they are suitable for handling short-term volatile tasks and do not have an advantage for long-term dependent tasks.
Hence, researchers have begun exploring the application of hybrid time series and deep learning models in epidemic prediction.Jin et al. (2022) proposed a parallelized approach based on a regression coefficient-weighted ARIMA-LSTM model to forecast epidemic data in China.They concluded that the epidemic in China would exhibit a steady downward trend over the next 60 days.Jin et al. (2023) modeled and forecasted the pandemic in Quebec, Canada, using a hybrid LSTM and ARIMA model.Through comparisons of three combined forecasting models, they demonstrated that CNN-ARIMA-LSTM exhibited the best prediction performance.These methods have been proven to have higher accuracy than single LSTM or ARIMA prediction models.Furthermore, we are preparing to enhance the prediction accuracy of the ARIMA-LSTM model through the integration of multidimensional data fusion.

Multi-source data
The data sources for pandemic prediction tasks are often diverse, encompassing case data, sentiment text data, and travel mobility data.Wang et al. (2020) utilized case data to perform time series predictions of the pandemic trends in Russia, Peru, and Iran.Chakraborty & Ghosh (2020) developed a novel hybrid autoregressive integrated moving average-wavelet-based forecasting (ARIMA-WBF) model using case data from multiple countries, including Canada, France, India, South Korea, and the UK, to predict and assess risks.While case data can directly forecast the development trends of epidemics, the limited quantity of data poses constraints on the accuracy of predictions.In addition, Alamoodi et al. (2021) conducted impact analysis and opinion mining on epidemic and other infectious diseases through sentiment text data on social media.Dubey ( 2020) performed nationallevel sentiment analysis based on Twitter comments during the pandemic.Sentiment text data is commonly employed for analyzing the societal impact of pandemics, and its reflection on the epidemic exhibits a lag effect.Alessandretti ( 2022) estimated real-world large-scale contact networks and developed scenario predictions to understand the driving factors of disease spread through mobile GPS data.Gan et al. (2020) assessed risks during the pandemic by collecting mobile traffic data.Zhan et al. (2020) modeled and predicted epidemic based on population migration data.These data cannot directly reflect the development of the pandemic.Yet, they can serve as supplementary for assessment and prediction.According to Kong et al. (2022), there is a significant correlation between the pandemic and population mobility.Therefore, we will design a method that integrates case data and population mobility data to enhance the accuracy of our prediction model.

PROPOSED METHOD
This section will provide a detailed description of the proposed methodology.Our approach involves utilizing an ARIMA-LSTM model designed for predicting multi-source epidemic data.Firstly, we employ the ARIMA-LSTM model to forecast the development trends of epidemic case data and population flow data.Secondly, we introduced a fully connected layer based on the Bayes-Attention mechanism to integrate predictions from multi-source data.Finally, we utilize various evaluation metrics to assess the prediction performance.Specifically, we will elaborate on the following four aspects: the ARIMA-LSTM model, the FC layer, and evaluation indicators.The architecture of the proposed model is shown in Fig. 2.

ARIMA-LSTM model
The ARIMA-LSTM model is used to predict the epidemic development trend.To address the long-term dependency and dynamic complexity issues in epidemic spread, we consider integrating traditional mathematical models and deep learning models to enhance prediction accuracy.Firstly, we segment the time series into low-volatility and high-volatility sequences using a moving average method.Secondly, we stabilize the sequence through differencing integration operations (I) and predict future values using an autoregressive moving average (ARMA) model for the low-volatility time series.Thirdly, we forecast future values using a long short-term memory (LSTM) model for the high-volatility time series.Finally, we add the prediction results of high-volatility and low-volatility sequences, resulting in the final sequence of case predictions.
Data partition: The moving average method can segment time series into low-volatility and high-volatility sequences involving simple moving average (SMA), exponential moving average (EMA), and weighted moving average (WMA).SMA excels in handling sequences with consistent weight coefficients.WMA is proficient in dealing with sequences where weight coefficients vary linearly with time intervals.EMA is adept at managing sequences where weight coefficients exhibit exponential changes over time intervals.We will assess the suitability of the current moving average method based on error values, thereby selecting the optimal moving average method.The calculation of the moving average method is as follows: In the above equations, N denotes the size of the moving average window, and P t −i represents the value at time t −i.w i signifies the weight coefficient, α denotes the smoothing factor, typically within the range of (0,1).ARIMA model: Autoregressive integrated moving average model, commonly known as ARIMA(p, d, q), involves parameters p, d, and q, representing the order of the autoregressive (AR) terms, the order of differencing (I) required to achieve stationarity in the time series, and the order of the moving average (MA) terms, respectively.The modeling and forecasting process of the ARIMA model is illustrated in Fig. 3.The ARIMA model forecasts future values by linearly calculating past observed values and time error terms of time series data.Therefore, the ARIMA model is particularly advantageous for data with clear linear trends and periodicity, making it suitable for predicting seasonal cyclic variations in epidemics.
Differential Integration (I) operation can transform a non-stationary sequence into a stationary one, typically involving first-order or higher-order differences.The order of differencing integration d can be determined through visualization methods by selecting the order that minimizes the fluctuations in the time series.Autoregression (AR) term expresses the relationship between current and historical values.It is used to predict the current value based on its past values.The formula for a p-order autoregressive process is defined as follows: where y t is the current value, µ is a constant term, p is the order, θ i represents the autoregressive coefficients, and ε t is the error term.
Moving average (MA) term represents the relationship between the current value and random errors, focusing on the accumulation of error terms in the autoregressive model.It effectively eliminates random fluctuations in predictions.The formula for a q-order moving average process is calculated as where y t is the current value, µ is a constant term, q is the order, θ i represents the moving average coefficients, and ε is the error term.Therefore, we can express the computational formula for ARIMA(p,d,q) as follows: To determine the optimal values of p and q, we computed the Bayesian Information Criterion (BIC).The BIC calculates the probability function and adds a penalty term for the number of parameters, which helps to avoid overfitting and provides a balanced approach to model selection.The calculation formula is described as where k is the number of parameters, n is the sample size, L is the likelihood function.
A smaller BIC value indicates that the given parameters provide a more accurate model description.4.This structure enables the LSTM model to dynamically learn and adjust its internal parameters to adapt to different time series data patterns and features.The LSTM model can retain relevant information over extended periods, making it suitable for predicting complex and evolving tasks during epidemic outbreaks.Next, we will proceed to offer an in-depth introduction to the gated structure of the LSTM model: The forget gate determines which information from the previous time step's cell state should be forgotten or discarded.It utilizes a sigmoid activation function to produce an output between 0 and 1, where 0 signifies complete forgetting, and 1 indicates complete retention.This aids the network in forgetting irrelevant information, mitigating gradient exploding and vanishing issues.The calculation formula for the forget gate F t is calculated as: where σ is the sigmoid activation function, W f is the weight matrix for the forget gate, h t −1 is the previous time step's hidden state, x t is the input at the current time step, and b f is the bias term for the forget gate.
The input gate regulates the extent to which new input data flows in.It incorporates a sigmoid activation function and its output ranges between 0 and 1.The role of the input gate is to determine which information should be updated and added to the cell state.If the output of the input gate is close to 1, the network considers this information necessary and should be incorporated into the cell state.The calculation process for the input gate is as follows: In the above equation, c t is the candidate cell state, W c is the weight matrix for the cell state, and b c is the bias term for the cell state.i t is the output of the input gate, W i is the weight matrix for the input gate, and b i is the bias term for the input gate.
The output gate regulates the output from the cell to the hidden state.It employs a sigmoid activation function to determine which parts of the cell state should be output.Simultaneously, it uses a tanh activation function to ensure that the output values h t are within the range of −1 to 1.The role of the output gate is to generate the final output of the LSTM unit based on the current information in the cell state.The calculation process is as follows: In the above equation, o t is the output gate's output, W o is the weight matrix for the output gate, b o is the bias term, and h t is the final output.Finally, we add the prediction results of the high-volatility series and the low-volatility series to get the final case prediction series.
where x t represents the forecast result of the low-volatility ARIMA model, and h t represents the forecast result of the high-volatility LSTM model.

FC layer
In the fully connected layer, we introduce a Bayes-Attention fusion mechanism.This mechanism can dynamically allocate attention weights among multiple data sources and integrate auxiliary data sources into the case data.Specifically, we first dynamically allocate attention weights among various data sources through the attention mechanism.Subsequently, based on these attention weights, we integrate auxiliary data sources into the case data through a Bayesian network.Let Y = {Y 1 ,Y 2 ,...,Y T } be the connection vector generated from multi-source data.The calculation formula for the Bayesian attention fusion mechanism is as follows: In the above equation, w represents the learned parameter vector, which maps the hidden states to attention weights.α denotes the attention weights, and P signifies the conditional probability distribution.Finally, to facilitate comparisons with other baseline and state-of-the-art models in the experimental section, We process the fused dataset through a softmax function to concentrate its values within the range [0,1].

Evaluation indicators
To compare the prediction effects of different models more intuitively, we use the rootmean-squared error (RMSE), the mean absolute percentage error (MAPE), the mean absolute error (MAE), and the r-squared score (R 2 ).RMSE, MAPE, and MAE are commonly used evaluation metrics based on the errors between actual and predicted values.R 2 is used to assess the goodness of fit of a model to the data.As our model not only needs to fit complex epidemic transmission scenarios but also maintain prediction accuracy, we will evaluate the model's goodness of fit to multi-source data using R 2 and assess the prediction accuracy using RMSE, MAPE, and MAE.RMSE: Root-mean-squared error (RMSE) is the square root of the mean of the squared differences between predicted and actual values.It demonstrates a remarkable sensitivity to prediction errors, exerting a more substantial penalty on significant mistakes than minor discrepancies.The more minor the RMSE, the smaller the gap between predicted and actual values, indicating a more accurate model.The formula for calculating RMSE is as follows: MAPE: Mean absolute percentage error (MAPE) represents the average percentage error of each observed value.It expresses errors in percentage form, making it more easily understandable.The smaller the MAPE, the smaller the prediction error, indicating higher model accuracy.The calculation formula for MAPE is as follows: MAE: Mean absolute error (MAE) is the average of the absolute differences between predicted and actual values.Unlike RMSE, MAE does not consider the square of errors, R 2 : R-squared (R 2 ) represents the extent to which the model explains the variability of the target variable.It evaluates how well a model fits the data.It ranges from 0 to 1, with values closer to 1 indicating a better fit of the model.An R 2 of 1 signifies a perfect fit to the data, while an R 2 of 0 implies that the model cannot explain the variability of the target variable.The calculation formulas for R 2 is as follows: In the above formulas, n is the total sample size, ŷi is the predicted value of the model, y i is the true value, ȳ is the average of the true value.

EXPERIMENTS Datasets
The experiments of our proposed model were completed in Python 3.11.6.This study utilizes the cumulative confirmed case numbers in New York State from March 2, 2020, to February 24, 2023, and daily population mobility trajectory numbers from February 15, 2020, to October 15, 2022, for modeling, prediction, and analysis.The data were sourced from the Johns Hopkins University website and the U.S. COVID-19 pandemic population mobility dataset, both publicly available.The datasets are shown in Table 1.
Upon acquiring the data, the outliers were cleaned through time series diagrams, and mean substitution treatment was applied for outlier handling.We utilized the first 80% of the data as the training set for the model and the remaining 20% as the test set to assess the model's generalization ability.

ARIMA-LSTM-FC model
This study initially trains the ARIMA-LSTM model using cumulative confirmed case data from New York, NY, USA.The hyperparameter training process for the model is depicted in Fig. 5. Due to MAPE representing the mean percentage error and exhibiting superior performance, we utilize MAPE to assess the training effectiveness of the model.Subsequently, future trend predictions are derived separately based on confirmed case data and population mobility data.Finally, the prediction outcomes of population mobility data are integrated into the case data through a fully connected (FC) layer utilizing the Bayes-Attention mechanism.
The parameters for the ARIMA-LSTM-FC model are configured as specified in Table 2.The time series graph comparing the predicted values is shown in Fig. 6.

Comparison experiment
The ARIMA-LSTM-FC hybrid model exhibits better prediction accuracy based on the model prediction graphs.We introduce several baseline and advanced models to provide a more precise comparison of the prediction performance.These time series forecasting models include LSTM, Bi-LSTM, GRU, Transformer, and CNN-ARIMA-LSTM.
LSTM: Shahid, Zameer & Muneeb (2020) utilized LSTM, GRU, and Bi-LSTM models to forecast the time series of confirmed cases, deaths, and recoveries in 10 major countries.We implement the model methodologies from their work and perform prediction tasks based on our dataset.Bi-LSTM: The bidirectional long short-term memory (Shahid, Zameer & Muneeb, 2020) (Bi-LSTM) is an extension of the LSTM network designed for processing sequential data.It introduces two independent LSTM layers at each time step: one for forward propagation and the other for backward propagation.Unlike the standard LSTM network, Bi-LSTM can consider both past and future information of sequence data simultaneously, enabling a more comprehensive understanding and modeling of the context of sequential data.

GRU:
The gated recurrent unit (GRU) (Shahid, Zameer & Muneeb, 2020) is a variant of the RNN designed for processing sequential data.It controls the flow and update of information in sequence data through gate mechanisms, including update and reset gates, thereby modeling long-term dependencies.The GRU model has fewer parameters, faster training speed, and better generalization capabilities.
Transformer: The Transformer is a deep learning model based on attention mechanisms, with its core ideas including the self-attention mechanism and positional encoding.Its modular structure and parallel computation capabilities address the efficiency issues of RNNs when dealing with long sequences.
We evaluate the performance of each model using evaluation metrics such as the root-mean-squared error (RMSE), the mean absolute percentage error (MAPE), the mean absolute error (MAE), and the determination coefficient (R 2 ).The prediction performance evaluation metrics of each model on the test set are shown in Table 3.
Based on the four evaluation metrics selected in this study, we compare the performance of various models.We denote the best-performing metric in bold and the second-best metric underlined.Among them, the ARIMA-LSTM-FC model demonstrates the best prediction performance, followed by the CNN-ARIMA-LSTM model.Comparative

Ablation study
To validate that the hybrid model performs better than individual models, we conducted ablation experiments through the following three parts: the ARIMA model, the LSTM model, and the ARIMA-LSTM model.As our fully connected layer is intended to incorporate the prediction outcomes from auxiliary data sources into the case data, the resulting trend still reflects the future trajectory of the case data.We will conduct ablation experiments based on case data to ensure a consistent dimension for comparing experimental results.ARIMA model: We prioritize the model training using the cumulative confirmed cases.Initially, the original data undergoes differencing.After applying a second-order difference, the signal sequence becomes stationary and white noise-free.The parameters (p, d, q) of the model and the white noise nature of the residual were determined using the Bayesian information criterion (BIC).A smaller BIC value indicates a more precise model description based on the parameters.The calculation process is illustrated in Eq. ( 7).The results of the BIC are depicted in a heatmap, as shown in Fig. 7.
The red mark on the heatmap shows that the final is ARIMA (2,2,2).Finally, the comparison between the ARIMA model's predicted values and the actual values in the time series is depicted in Fig. 8.
LSTM model: The LSTM model is constructed using the cumulative confirmed cases in New York State from March 2, 2020, to February 24, 2023.Both the input and output of the model are cumulative confirmed case numbers.To determine the parameters of the LSTM model, we conducted hyperparameter experiments, as shown in Fig. 5.
Finally, the model parameters are configured as specified in Table 4.
Based on the specified parameters, we construct the LSTM model.The time series graph comparing the predicted values of the LSTM model with the actual values is shown in Fig. 9.
Once the parameters are configured, we train the ARIMA-LSTM model using the training set and then utilize the trained model to make predictions on the test set.The time series graph comparing the predicted values of the ARIMA-LSTM model with the actual values is illustrated in Fig. 10.
According to the comparative analysis of prediction results from various models, it is evident that the ARIMA-LSTM-FC model indeed exhibits superior forecasting accuracy.We still use evaluation metrics such as the root mean square error (RMSE), the mean absolute percentage error (MAPE), the mean absolute error (MAE), and the determination coefficient (R 2 ) to make a more precise comparison.The performance evaluation metrics for each model on the test set are presented in Table 6.
Based on four evaluation criteria, we compare the performance of our proposed models.Here, we denote the best-performing metric with bold and the second-best metric with underline.Among them, the ARIMA-LSTM-FC model demonstrates the best prediction performance, with ARIMA-LSTM following closely.Since the ARIMA-LSTM-FC model integrates multiple data sources, its performance surpasses that of the ARIMA-LSTM model.Moreover, due to the combination of linear and nonlinear methods, the prediction performance of the ARIMA-LSTM model exceeds that of standalone LSTM or ARIMA models.Through the fully connected layer, we integrate the prediction outcomes from auxiliary data sources into the case data.This approach ensures that both the input and output of the model consist of case data, which is crucial for the effectiveness of ablation experiments.The results of ablation experiments indicate that the hybrid prediction model and multi-source data fusion indeed enhance prediction performance.

CONCLUSION
The transmission of epidemics exhibits periodic dependency and dynamic complexity characteristics, posing significant challenges to forecasting efforts.We propose an ARIMA-LSTM model based on multi-source data fusion to predict the epidemic's development trends more accurately.Initially, we employ the ARIMA-LSTM model to forecast trends in multiple data sources separately.Subsequently, we introduce a fully connected (FC) layer based on a Bayesian attention mechanism to integrate the prediction outcomes from auxiliary data sources into the case data.The computational metrics for the ARIMA-LSTM-FC model are as follows: RMSE = 0.022, MAPE = 0.42, MAE = 0.032, R 2 = 0.993.Among them, all these metrics outperform those of other compared models.Experimental results demonstrate that the prediction performance of the ARIMA-LSTM-FC model surpasses that of the ARIMA-LSTM model, and the ARIMA-LSTM model's prediction performance exceeds that of the standalone LSTM or ARIMA models.Thus, the fusion of the ARIMA-LSTM model with multi-source data enhances prediction performance.
Our data fusion method, anchored in Bayesian networks and attention mechanisms, dynamically allocates attention weights for integrating multiple data sources without imposing any constraints on the data.The basic principle of the ARIMA-LSTM hybrid model lies in predicting future values based on historical data.Numerous studies have already explored the application of the ARIMA-LSTM model across various domains, demonstrating its versatility across different scenarios.Thus, the multi-source data fusionbased ARIMA-LSTM model we propose exhibits universal applicability.
The COVID-19 pandemic has surpassed the period covered in this article.However, the outbreak of the pandemic has undoubtedly sounded an alarm worldwide, emphasizing the need to invest in prediction tasks related to epidemic development trends.The multisource data-fused ARIMA-LSTM forecasting model established in this article can serve as a reference for governments of various countries engaged in similar infectious disease control efforts, aiming to minimize the losses caused by pandemics.In the future, we have plans to propose more efficient models for the fusion of multi-source data and explore predicting the development trends of epidemics in countries where case data is lacking.

Figure 1
Figure 1 Time series changes in confirmed cases and population flow data.Full-size DOI: 10.7717/peerjcs.2046/fig-1

Figure 8
Figure 8 Time series diagram comparing the predicted and true values of the ARIMA model.Full-size DOI: 10.7717/peerjcs.2046/fig-8

Figure 9
Figure 9 Time series diagram comparing the predicted and true values of the LSTM model.Full-size DOI: 10.7717/peerjcs.2046/fig-9

Wang et al. (2024), PeerJ Comput. Sci., DOI 10.7717/peerj-cs.2046 10/21 Table 1 Specific details of the dataset. The cumulative confirmed case dataset
of large and small errors relatively equal.The smaller the MAE, the smaller the absolute difference between the actual and predicted values, indicating a more accurate model prediction.The formula for calculating MAE is as follows:

Figure 10 Time series diagram comparing the predicted and true values of ARIMA-LSTM model. Full-size DOI: 10.7717/peerjcs.2046/fig-10 Table 6 Ablation experiment results.
Values in bold indicate the best-performing metric.An underline indicates the second-best metric.